//
//  UIImage+Extension.swift
//  MySinaWeibo
//
//  Created by eastriver on 17/3/28.
//  Copyright © 2017年 believewilliam. All rights reserved.
//

import UIKit

extension UIImage {
    class func snapShotForCurrentScreen() -> UIImage? {
       
        let window = UIApplication.shared.keyWindow!
        // 开启图形上下文
        UIGraphicsBeginImageContextWithOptions(window.bounds.size, true, UIScreen.main.scale)
        // 将window绘制到图形上下文中
        window.drawHierarchy(in: window.bounds, afterScreenUpdates: false)
        // 从图形上下文中获取image
        let currentImage = UIGraphicsGetImageFromCurrentImageContext()
        // 关闭图形上下文
        UIGraphicsEndImageContext()
        return currentImage
        
        /*
        // 1. 开启图片上下文
        let window = UIApplication.shared.keyWindow!

        UIGraphicsBeginImageContextWithOptions(window.bounds.size, false, UIScreen.main.scale)
        
        // 2. 获取到的上下文 是 bitMap 上下文
        let context = UIGraphicsGetCurrentContext()
        
        // 3. 绘制图形
        context?.move(to: CGPoint(x: 0, y: 0))
            context?.addLine(to: CGPoint(x: 100, y: 100))
        // 4. 渲染
        context?.strokePath()
        // 5. 从当前上下文中, 获取图片
        // 需要在 关闭之前获取图片
        let clipImage = UIGraphicsGetImageFromCurrentImageContext()
        
        // 6. 到这个位置, 就意味着 图片上下文已经使用完成
        // 要关闭 图片上下文
        UIGraphicsEndImageContext();
        return clipImage
 */
    }
}
